import os
import logging
from logging.handlers import RotatingFileHandler
from datetime import datetime

# 日志级别映射
LOG_LEVELS = {
    'debug': logging.DEBUG,
    'info': logging.INFO,
    'warning': logging.WARNING,
    'error': logging.ERROR,
    'critical': logging.CRITICAL
}

def setup_logger(name='chatgpt_app', level='info', log_file=None):
    """设置并返回一个配置好的日志记录器
    
    Args:
        name (str): 日志记录器名称
        level (str): 日志级别 ('debug', 'info', 'warning', 'error', 'critical')
        log_file (str, optional): 日志文件路径。如果为None，将使用默认路径
        
    Returns:
        logging.Logger: 配置好的日志记录器
    """
    # 获取日志级别
    log_level = LOG_LEVELS.get(level.lower(), logging.INFO)
    
    # 创建日志记录器
    logger = logging.getLogger(name)
    logger.setLevel(log_level)
    
    # 如果已经有处理器，不再添加
    if logger.handlers:
        return logger
    
    # 创建格式化器
    formatter = logging.Formatter(
        '%(asctime)s - %(name)s - %(levelname)s - %(message)s'
    )
    
    # 添加控制台处理器
    console_handler = logging.StreamHandler()
    console_handler.setLevel(log_level)
    console_handler.setFormatter(formatter)
    logger.addHandler(console_handler)
    
    # 如果指定了日志文件，添加文件处理器
    if log_file is None:
        # 创建logs目录（如果不存在）
        logs_dir = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'logs')
        os.makedirs(logs_dir, exist_ok=True)
        
        # 默认日志文件名：logs/chatgpt_app_YYYY-MM-DD.log
        today = datetime.now().strftime('%Y-%m-%d')
        log_file = os.path.join(logs_dir, f'chatgpt_app_{today}.log')
    
    # 创建文件处理器（最大10MB，保留5个备份）
    file_handler = RotatingFileHandler(
        log_file, maxBytes=10*1024*1024, backupCount=5
    )
    file_handler.setLevel(log_level)
    file_handler.setFormatter(formatter)
    logger.addHandler(file_handler)
    
    return logger

# 创建默认日志记录器
logger = setup_logger()

def get_logger(name=None):
    """获取指定名称的日志记录器
    
    如果未指定名称，返回默认日志记录器
    
    Args:
        name (str, optional): 日志记录器名称
        
    Returns:
        logging.Logger: 日志记录器
    """
    if name is None:
        return logger
    return logging.getLogger(name)